package com.example.easyexceldemo.controller;
 
import com.alibaba.excel.EasyExcel;
import com.alibaba.excel.util.ListUtils;
import com.alibaba.excel.util.MapUtils;
import com.alibaba.fastjson2.JSON;
import com.example.easyexceldemo.entity.DownloadData;
import com.example.easyexceldemo.entity.ResultObj;
import com.example.easyexceldemo.entity.UploadData;
import com.example.easyexceldemo.handler.UploadDataListener;
import lombok.RequiredArgsConstructor;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
 
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.net.URLEncoder;
import java.util.*;
import java.util.stream.Collectors;

@RestController
@RequestMapping("/excel")
@RequiredArgsConstructor
public class ExcelController {

    private final StringRedisTemplate redisTemplate;

//    @Autowired
//    private UploadDAO uploadDAO;
 
    /**
     * 文件下载（失败了会返回一个有部分数据的Excel）
     * <p>
     * 1. 创建excel对应的实体对象 参照{@link }
     * <p>
     * 2. 设置返回的 参数
     * <p>
     * 3. 直接写，这里注意，finish的时候会自动关闭OutputStream,当然你外面再关闭流问题不大
     */
    @GetMapping("download")
    public void download(HttpServletResponse response) throws IOException {
        // 这里注意 有同学反应使用swagger 会导致各种问题，请直接用浏览器或者用postman
        response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");
        response.setCharacterEncoding("utf-8");
        // 这里URLEncoder.encode可以防止中文乱码 当然和easyexcel没有关系
        String fileName = URLEncoder.encode("测试", "UTF-8").replaceAll("\\+", "%20");
        response.setHeader("Content-disposition", "attachment;filename*=utf-8''" + fileName + ".xlsx");
//        List<List<String>> head = new ArrayList<>();
        EasyExcel.write(response.getOutputStream(), DownloadData.class)
                .sheet("入职模版")
                .doWrite(data(true));
    }
 
    /**
     * 文件下载并且失败的时候返回json（默认失败了会返回一个有部分数据的Excel）
     *
     * @since 2.1.1
     */
    @GetMapping("downloadFailedUsingJson")
    public void downloadFailedUsingJson(HttpServletResponse response) throws IOException {
        // 这里注意 有同学反应使用swagger 会导致各种问题，请直接用浏览器或者用postman
        try {
            response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");
            response.setCharacterEncoding("utf-8");
            // 这里URLEncoder.encode可以防止中文乱码 当然和easyexcel没有关系
            String fileName = URLEncoder.encode("测试", "UTF-8").replaceAll("\\+", "%20");
            response.setHeader("Content-disposition", "attachment;filename*=utf-8''" + fileName + ".xlsx");
            // 这里需要设置不关闭流
            EasyExcel.write(response.getOutputStream()).autoCloseStream(Boolean.FALSE).sheet("模板")

                    .doWrite(data(false));
        } catch (Exception e) {
            // 重置response
            response.reset();
            response.setContentType("application/json");
            response.setCharacterEncoding("utf-8");
            Map<String, String> map = MapUtils.newHashMap();
            map.put("status", "failure");
            map.put("message", "下载文件失败" + e.getMessage());
            response.getWriter().println(JSON.toJSONString(map));
        }
    }
 
    /**
     * 文件上传
     * <p>
     * 1. 创建excel对应的实体对象 参照{@link UploadData}
     * <p>
     * 2. 由于默认一行行的读取excel，所以需要创建excel一行一行的回调监听器，参照{@link UploadDataListener}
     * <p>
     * 3. 直接读即可
     */
    @PostMapping("/upload")
    @ResponseBody
    public Object upload(@RequestParam("file") MultipartFile file) throws IOException {
        long time = new Date().getTime();
        String lotId = String.format("%s_%s_%S", 11,22, time);
        System.out.println(lotId);
        EasyExcel.read(file.getInputStream(), UploadData.class, new UploadDataListener("hello world",redisTemplate)).sheet().doRead();

        String totalRowNumber = redisTemplate.opsForValue().get("hello world");
        Map<Object, Object> fail = redisTemplate.opsForHash().entries("hello world" + "fail");
        List<String> strings = redisTemplate.opsForList().range("hello world" + "success", 0,-1);
        List<UploadData> reuslt = strings.stream().map(m -> {
            UploadData uploadData = JSON.parseObject(m, UploadData.class);
            return uploadData;
        }).collect(Collectors.toList());
        ResultObj resultObj = new ResultObj();
        resultObj.setFail(fail);
        resultObj.setSuccess(reuslt);
        resultObj.setTotalNumber(totalRowNumber);
        return resultObj;
    }
 
    private List<DownloadData> data(boolean flag) {
        List<DownloadData> list = ListUtils.newArrayList();
        DownloadData downloadData = new DownloadData("张三","19912345678","综合部","文员","普通员工","2024/11/20","全职","试用");
        DownloadData downloadData1 = new DownloadData("张三2","asdf","hah ","垃圾","？？？","2024/11/20","呵呵","already");
        list.add(downloadData);
        list.add(downloadData1);
        return list;
    }
}
